#pragma once

#include <string>
#include <vector>
#include <sys/types.h>
#include <iostream>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <atomic>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp/";

    class PathUtil
    {
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        // 编译时需要的临时文件
        // 构建源文件的路径+后缀的完整文件名
        // abc -> abc.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 构建可执行文件的完整路径+后缀
        // abc -> abc.exe
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        // 构建程序对应的编译错误完整路径+后缀
        // abc -> abc.compile_error
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 运行时需要的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        // 构建程序对应的标准完整路径+后缀
        // abc -> abc.stderr
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };


    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }
        //获得毫秒级时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec*1000 + _time.tv_usec/1000);
        }
        
    };
    class FileUtil
    {
    public:
        static bool IsFileExists(const std::string &file_name)
        {
            // stat函数可以查看文件属性
            struct stat st;
            if (stat(file_name.c_str(), &st) == 0)
            {
                return true;
            }
            return false;
        }

        static std::string UniqFileName()
        {
            static std::atomic_uint id(0);
            //毫秒级时间戳+原子性递增, 保证唯一性
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);
            id++;

            return ms + "_" + uniq_id;  
        }

        static bool WriteFile(const std::string &target, const std::string &code)
        {   
            std::ofstream out(target);
            if(!out.is_open())
                return false;
            out.write(code.c_str(), code.size());
            out.close();
            return true;
        }

        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(target);
            if(!in.is_open())
            {
                return false;
            }
            std::string line;
            //getline: 不保存分隔符, 有些时候需要保存\n, 添加keep参数, 判断是否保留\n
            //getline 内部重载了强制类型转换
            while(std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };
    class StringUtil
    {
    public:
    /***********************************
     * str: 输入要切分的字符串
     * out: 输出切分完毕的字符串数组
     * sep: 分割符
    ***********************************/
        static void SplitString(const std::string &str, std::vector<std::string> *out, const std::string& sep)
        {   
            //boost split
            boost::split((*out), str, boost::is_any_of(sep),boost::algorithm::token_compress_on);
        }   
    };

}



